﻿/*
 * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
 *
 * This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
 *
 * Use of this source code is governed by MIT license that can be found in the
 * LICENSE file in the root of the source tree. All contributing project authors
 * may be found in the AUTHORS file in the root of the source tree.
 */

#ifndef ZLMEDIAKIT_H265_H
#define ZLMEDIAKIT_H265_H

#include "Frame.h"
#include "Track.h"
#include "Util/base64.h"
#include "H264.h"
#define H265_TYPE(v) (((uint8_t)(v) >> 1) & 0x3f)
using namespace toolkit;

namespace mediakit {

bool getHEVCInfo(const string &strVps, const string &strSps, int &iVideoWidth, int &iVideoHeight, float &iVideoFps);

/**
 * 265帧类
 */
class H265Frame : public FrameImp {
public:
    using Ptr = std::shared_ptr<H265Frame>;
    enum {
        NAL_TRAIL_N = 0,
        NAL_TRAIL_R = 1,
        NAL_TSA_N = 2,
        NAL_TSA_R = 3,
        NAL_STSA_N = 4,
        NAL_STSA_R = 5,
        NAL_RADL_N = 6,
        NAL_RADL_R = 7,
        NAL_RASL_N = 8,
        NAL_RASL_R = 9,
        NAL_BLA_W_LP = 16,
        NAL_BLA_W_RADL = 17,
        NAL_BLA_N_LP = 18,
        NAL_IDR_W_RADL = 19,
        NAL_IDR_N_LP = 20,
        NAL_CRA_NUT = 21,
        NAL_RSV_IRAP_VCL22 = 22,
        NAL_RSV_IRAP_VCL23 = 23,

        NAL_VPS = 32,
        NAL_SPS = 33,
        NAL_PPS = 34,
        NAL_AUD = 35,
        NAL_EOS_NUT = 36,
        NAL_EOB_NUT = 37,
        NAL_FD_NUT = 38,
        NAL_SEI_PREFIX = 39,
        NAL_SEI_SUFFIX = 40,
    };
    static const char * GetFrameName(int nalu_type) {
        switch(nalu_type){
            case NAL_TRAIL_N:
                return "NAL_TRAIL_N";
            case NAL_TRAIL_R:
                return "NAL_TRAIL_R";
            case NAL_TSA_N:
                return "NAL_TSA_N";
            case NAL_TSA_R:
                return "NAL_TSA_R";
            case NAL_STSA_N:
                return "NAL_STSA_N";
            case NAL_STSA_R:
                return "NAL_STSA_R";            
            case NAL_RADL_N:
                return "NAL_RADL_N";            
            case NAL_RADL_R:
                return "NAL_RADL_R";            
            case NAL_RASL_N:
                return "NAL_RASL_N";            
            case NAL_RASL_R:
                return "NAL_RASL_R";            
            case NAL_BLA_W_LP:
                return "NAL_BLA_W_LP";            
            case NAL_BLA_W_RADL:
                return "NAL_BLA_W_RADL";                      
            case NAL_BLA_N_LP:
                return "NAL_BLA_N_LP";                      
            case NAL_IDR_W_RADL:
                return "NAL_IDR_W_RADL";                      
            case NAL_IDR_N_LP:
                return "NAL_IDR_N_LP";                      
            case NAL_CRA_NUT:
                return "NAL_CRA_NUT";                      
            case NAL_RSV_IRAP_VCL22:
                return "NAL_RSV_IRAP_VCL22";                      
            case NAL_RSV_IRAP_VCL23:
                return "NAL_RSV_IRAP_VCL23";
            case NAL_VPS:
                return "NAL_VPS";
            case NAL_SPS:
                return "NAL_SPS";
            case NAL_PPS:
                return "NAL_PPS";
            case NAL_AUD:
                return "NAL_AUD";
            case NAL_EOS_NUT:
                return "NAL_EOS_NUT";
            case NAL_EOB_NUT:
                return "NAL_EOB_NUT";            
            case NAL_FD_NUT:
                return "NAL_FD_NUT";            
            case NAL_SEI_PREFIX:
                return "NAL_SEI_PREFIX";            
            case NAL_SEI_SUFFIX:
                return "NAL_SEI_SUFFIX";
            default:
                return "Unknown NAL";
        }
    }
    bool keyFrame() const override;
    bool configFrame() const override;
    static bool isKeyFrame(int type);

protected:
    friend class FrameImp;
    friend class ResourcePool_l<H265Frame>;
    H265Frame();
};

class H265FrameNoCacheAble : public FrameFromPtr {
public:
    using Ptr = std::shared_ptr<H265FrameNoCacheAble>;

    H265FrameNoCacheAble(char *ptr, size_t size, uint32_t dts,uint32_t pts, size_t prefix_size/* = 4*/,time_t playStartTime);
    bool keyFrame() const override;
    bool configFrame() const override;
};

/**
* 265视频通道
*/
class H265Track : public VideoTrack {
public:
    using Ptr = std::shared_ptr<H265Track>;

    /**
     * 不指定sps pps构造h265类型的媒体
     * 在随后的inputFrame中获取sps pps
     */
    H265Track() = default;

    /**
     * 构造h265类型的媒体
     * @param vps vps帧数据
     * @param sps sps帧数据
     * @param pps pps帧数据
     * @param vps_prefix_len 265头长度，可以为3个或4个字节，一般为0x00 00 00 01
     * @param sps_prefix_len 265头长度，可以为3个或4个字节，一般为0x00 00 00 01
     * @param pps_prefix_len 265头长度，可以为3个或4个字节，一般为0x00 00 00 01
     */
    H265Track(const string &vps,const string &sps, const string &pps,int vps_prefix_len = 4, int sps_prefix_len = 4, int pps_prefix_len = 4);

    /**
     * 返回不带0x00 00 00 01头的vps/sps/pps
     */
    const string &getVps() const;
    const string &getSps() const;
    const string &getPps() const;

    bool ready() override;
    CodecId getCodecId() const override;
    int getVideoWidth() const override;
    int getVideoHeight() const override;
    float getVideoFps() const override;
    void inputFrame(const Frame::Ptr &frame) override;

private:
    void onReady();
    Sdp::Ptr getSdp() override;
    Track::Ptr clone() override;
    void inputFrame_l(const Frame::Ptr &frame);
    void insertConfigFrame(const Frame::Ptr &frame);

private:
    bool _is_idr = false;
    int _width = 0;
    int _height = 0;
    float _fps = 0;
    string _vps;
    string _sps;
    string _pps;
};

}//namespace mediakit
#endif //ZLMEDIAKIT_H265_H
